package com.selector;

import java.util.List;
import java.util.Scanner;

public class ModelSelector {
    private static final List<String> MODELS = List.of(
            "deepseek-r1:14b",
            "deepseek-r1:7b",
            "qwen2.5:3b"
    );
    /*
            "deepseek-r1:1.5b",
            "deepseek-r1:7b",
            "deepseek-r1:8b",
            "deepseek-r1:14b",
            "deepseek-r1:32b",
            "deepseek-r1:70b",
            "unsloth_deepseek-r1:7B",
            "thesis:8b",
            "thesis:70b",
            "thesis:14b",
            "thesis:32b"
     */


    public String selectModel() {
        Scanner scanner = new Scanner(System.in);
        for (int i = 0; i < MODELS.size(); i++) {
            System.out.printf("%d、%s%n", i + 1, MODELS.get(i));
        }
        System.out.print("选择模型:");
        int choice = Integer.parseInt(scanner.nextLine().trim());
        return MODELS.get(choice - 1);
    }
}